#ifndef DATASTATION_H_
#define DATASTATION_H_

/*
 * @Author Hien Hoang
 * @Version Nov 28, 2013
 * */

#include <map>
#include "IPv4Address.h"
#include "NotificationBoard.h"
#include "DATAPacket_m.h"

class DataStation : public cSimpleModule {

protected:
    typedef std::map<IPv4Address, int> PacketCount;
    PacketCount numReceived;
    int numReceivedSum;
    NotificationBoard* nb;

public:
    virtual void initialize();
    virtual void handleMessage(cMessage *msg);
    virtual void handleIncomingPacket(DataPacket *pkt);
    virtual void updateNodeDisplay();

};


#endif /* DATASTATION_H_ */
